"""
和游戏窗口操作相关的模块
"""
import pygame
from window import const

# 窗口对象
win: pygame.Surface | None = None


def init_win():
    """
    初始化游戏窗口
    :return:
    """
    global win
    # 创建游戏窗口
    win = pygame.display.set_mode(const.WIN_SIZE)
    # 设置游戏窗口的标题
    pygame.display.set_caption(const.WIN_TITLE)


def get_win() -> pygame.Surface:
    """
    获取窗口对象
    :return:
    """
    global win
    return win
